draw spiral in python